# mmdeploy/configs/mmdet/detection/detection_ascend_static-640x640.py

codebase_config = dict(
    type='mmdet',
    task='ObjectDetection',
    model_type='end2end',
    post_processing=dict(
        score_threshold=0.05,
        confidence_threshold=0.005,
        iou_threshold=0.5,
        max_output_boxes_per_class=200,
        pre_top_k=5000,
        keep_top_k=100,
        background_label_id=-1
    )
)

onnx_config = dict(
    type='onnx',
    export_params=True,
    keep_initializers_as_inputs=False,
    opset_version=11,
    input_names=['input'],
    output_names=['dets', 'labels'],
    input_shape=(800, 1088),
    optimize=False
)

backend_config = dict(
    type='ascend',
    soc_version='Ascend310P3',
    model_inputs={
        'input_shapes': {'input': (800, 1088)},
        'batch_sizes': [1]
    }
)
